import { isFunction } from "@/utils";
import axios, { type AxiosRequestConfig, type Canceler } from "axios";
import qs from "qs";

// 声明一个 Map 用于存储每个请求的标识 和 取消函数
let pendingMap = new Map<string, Canceler>();

// 序列化参数
export const getPendingUrl = (config: AxiosRequestConfig) =>
  [config.method, config.url, qs.stringify(config.data), qs.stringify(config.params)].join("&");

export class AxiosCanceler {
  /**
   * @description 添加请求
   * @param {Object} config
   * @return void
   */
  addPendingRequest(config: AxiosRequestConfig) {
    // 在请求开始前，对之前的请求做检查取消操作
    this.removePendingRequest(config);
    const url = getPendingUrl(config);
    config.cancelToken =
      config.cancelToken ||
      new axios.CancelToken(cancel => {
        if (!pendingMap.has(url)) {
          // 如果 pending 中不存在当前请求，则添加进去
          pendingMap.set(url, cancel);
        }
      });
  }

  /**
   * @description 移除请求
   * @param {Object} config
   */
  removePendingRequest(config: AxiosRequestConfig) {
    const url = getPendingUrl(config);

    if (pendingMap.has(url)) {
      // 如果在 pending 中存在当前请求标识，需要取消当前请求，并且移除
      const cancel = pendingMap.get(url);
      cancel && cancel();
      pendingMap.delete(url);
    }
  }

  /**
   * @description 清空所有pending
   */
  removeAllPendingRequest() {
    pendingMap.forEach(cancel => {
      cancel && isFunction(cancel) && cancel();
    });
    pendingMap.clear();
  }

  /**
   * @description 重置
   */
  reset(): void {
    pendingMap = new Map<string, Canceler>();
  }
}
